Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Added Atan2Op support #1056

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Added Atan2Op support #1056

wants to merge 2 commits into from

Conversation

kmitrovicTT
Copy link
Contributor

@kmitrovicTT kmitrovicTT commented Oct 24, 2024

Fixes #1204

Added Atan2Op to TTIR and TTNN ops. Added stablehlo to ttir conversion. Added ttir to ttnn conversion. Added tests.

…n. Added ttir to ttnn conversion. Added tests.
@kmitrovicTT
Copy link
Contributor Author

kmitrovicTT commented Oct 25, 2024

Seems like ttnn::atan2 reverses arguments compared to stablehlo/numpy/torch. Will check with ttnn crew why this is the case, and whether it should be fixed on their end, or if I should reverse args in ttir -> ttnn pass.

Blocked by #1113

@@ -1,6 +1,6 @@
include(ExternalProject)

set(TT_METAL_VERSION "047cdd97fa26b229208ce2f6f8b149c3df277be0")
set(TT_METAL_VERSION "fccd6929c7a065b9145af92b1a429b2e8f63907b")
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Not sure if this was intentional, or just got picked up during git add. Leaving comment as a reminder to remove before merging.

@kmitrovicTT kmitrovicTT added the MLIR Ops Issues related to MLIR dialect ops and their implementations label Nov 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
MLIR Ops Issues related to MLIR dialect ops and their implementations
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[Ops] Add support for TTIR/TTNN Atan2 op
3 participants